2024-01-05 分类：GO 作者：admin 阅读（27）
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
// 通过这个递归函数，可以确保结构体中所有未初始化的切片字段都被初始化为空切片，以便后续操作 // 把未初始化的数组默认值nil,初始为空切片，从而解决结构体转json时，nil转为null，而不是[] func InitSliceFields(data interface{}) { initSliceField(reflect.ValueOf(data).Elem()) } func initSliceField(value reflect.Value) { for i := 0; i < value.NumField(); i++ { fieldValue := value.Field(i) if fieldValue.Type().Kind() == reflect.Slice && fieldValue.IsNil() { fieldValue.Set(reflect.MakeSlice(fieldValue.Type(), 0, 0)) } else if fieldValue.Kind() == reflect.Struct { initSliceField(fieldValue) } else if fieldValue.Kind() == reflect.Ptr && fieldValue.Elem().Kind() == reflect.Struct { initSliceField(fieldValue.Elem()) } } } |
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
// 结构体属性指针迭代初始化 // 把结构体中属性中的指针，递归初始化，以防空指针 func InitStruct(obj interface{}) { value := reflect.ValueOf(obj).Elem() typ := value.Type() for i := 0; i < value.NumField(); i++ { fieldValue := value.Field(i) fieldType := typ.Field(i) if fieldValue.Kind() == reflect.Ptr && fieldValue.IsNil() { newFieldValue := reflect.New(fieldType.Type.Elem()) fieldValue.Set(newFieldValue) if fieldType.Type.Elem().Kind() == reflect.Struct { InitStruct(newFieldValue.Interface()) } } else if fieldValue.Kind() == reflect.Ptr && !fieldValue.IsNil() { // 如果属性是指针类型且不为nil，则判断其所指向的结构体是否可以转换为接口类型，并递归调用 InitStruct 初始化该结构体 if fieldValue.Elem().Kind() == reflect.Struct && fieldValue.Elem().Addr().CanInterface() { InitStruct(fieldValue.Elem().Addr().Interface()) } } else if fieldValue.Kind() == reflect.Struct { if fieldValue.Addr().CanInterface() { InitStruct(fieldValue.Addr().Interface()) } } } } |
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 |
// 按属性去重结构体数组 func DeduplicateStructs(slice interface{}, property string) interface{} { sliceValue := reflect.ValueOf(slice) if sliceValue.Kind() != reflect.Slice { panic("Input is not a slice") } elementType := sliceValue.Type().Elem() if elementType.Kind() != reflect.Struct { panic("Slice element is not a struct") } propertyField, found := elementType.FieldByName(property) if !found { panic("Property not found") } propertyIndex := propertyField.Index uniqueValues := make(map[interface{}]reflect.Value) resultSliceType := reflect.SliceOf(elementType) result := reflect.MakeSlice(resultSliceType, 0, 0) for i := 0; i < sliceValue.Len(); i++ { structValue := sliceValue.Index(i) propertyValue := structValue.FieldByIndex(propertyIndex) if _, ok := uniqueValues[propertyValue.Interface()]; !ok { uniqueValues[propertyValue.Interface()] = structValue result = reflect.Append(result, structValue) } } return result.Interface() } |
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 |
// 按属性去重结构体指针数组 func DeduplicateStructsPointer(slice interface{}, property string) interface{} { sliceValue := reflect.ValueOf(slice) if sliceValue.Kind() != reflect.Slice { panic("Input is not a slice") } elementType := sliceValue.Type().Elem() if elementType.Kind() != reflect.Ptr { panic("Slice element is not a pointer") } elementType = elementType.Elem() if elementType.Kind() != reflect.Struct { panic("Pointer element is not a struct") } propertyField, found := elementType.FieldByName(property) if !found { panic("Property not found") } propertyIndex := propertyField.Index uniqueValues := make(map[interface{}]reflect.Value) resultSliceType := reflect.SliceOf(reflect.PtrTo(elementType)) result := reflect.MakeSlice(resultSliceType, 0, 0) for i := 0; i < sliceValue.Len(); i++ { structValue := sliceValue.Index(i).Elem() propertyValue := structValue.FieldByIndex(propertyIndex) if _, ok := uniqueValues[propertyValue.Interface()]; !ok { uniqueValues[propertyValue.Interface()] = structValue result = reflect.Append(result, sliceValue.Index(i)) } } return result.Interface() } |
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
取得富文本中的图片地址 func OperRichText(texts []string) []string { var imgs []string for _, text := range texts { if text != "" { var docHtml, _ = goquery.NewDocumentFromReader(strings.NewReader(text)) sel := docHtml.Find("img") for i := range sel.Nodes { single := sel.Eq(i) src, flag := single.Attr("src") if flag && src != "" { imgs = append(imgs, src) } } } } return imgs } |
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 |
// 递归遍历结构体并设置空数据为初始值 便于返回json时候为空不返回，因为nil会被返回null func SetEmptyFieldsToZero(root interface{}) { // 获取根节点的反射值 rv := reflect.ValueOf(root) // 检查根节点是否为指针类型，如果是，获取其指向的值 if rv.Kind() == reflect.Ptr { rv = rv.Elem() } // 检查根节点是否为结构体类型 if rv.Kind() != reflect.Struct { //fmt.Println("Input is not a struct") return } // 遍历结构体的字段 for i := 0; i < rv.NumField(); i++ { fieldValue := rv.Field(i) // 检查字段的类型 switch fieldValue.Kind() { case reflect.Struct: // 如果字段是结构体类型，则递归调用本方法 SetEmptyFieldsToZero(fieldValue.Addr().Interface()) case reflect.Ptr: // 如果字段是指针类型 if fieldValue.IsNil() { // 如果指针为空，则创建一个新的实例并设置为字段的零值 fieldValue.Set(reflect.New(fieldValue.Type().Elem())) } else { // 如果指针不为空，则解引用指针并递归调用本方法 SetEmptyFieldsToZero(fieldValue.Interface()) } // 检查指针所指向的结构体中的字段是否都是零值，如果是，则将指针字段设置为 nil if fieldValue.Elem().Kind() == reflect.Struct { subStruct := fieldValue.Elem() isSubStructZero := true for j := 0; j < subStruct.NumField(); j++ { if !reflect.DeepEqual(subStruct.Field(j).Interface(), reflect.Zero(subStruct.Field(j).Type()).Interface()) { isSubStructZero = false break } } if isSubStructZero { fieldValue.Set(reflect.Zero(fieldValue.Type())) } } case reflect.Slice: // 如果字段是切片类型 if fieldValue.IsNil() { // 如果切片为空，则创建一个新的实例并设置为字段的零值 fieldValue.Set(reflect.MakeSlice(fieldValue.Type(), 0, 0)) } else { // 如果切片不为空，则遍历切片的每个元素并递归调用本方法 for j := 0; j < fieldValue.Len(); j++ { SetEmptyFieldsToZero(fieldValue.Index(j).Addr().Interface()) } } default: // 对于其他类型的字段，如果其值为空，则设置为零值 if reflect.DeepEqual(fieldValue.Interface(), reflect.Zero(fieldValue.Type()).Interface()) { fieldValue.Set(reflect.Zero(fieldValue.Type())) } } } } |
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 |
// 时间转换 func TimeStrToTime(timeStr string) (time.Time, error) { formats := []string{ "2006-01-02", "2006-01-02T15:04:05", "2006-01-02T15:04:05Z07:00", "2006-01-02T15:04:05.999Z07:00", "2006-01-02T15:04:05.999999Z07:00", "2006-01-02 15:04:05", "2006-01-02 15:04:05 -0700 MST", } var parsedTime time.Time var parseErr error for _, format := range formats { parsedTime, parseErr = time.Parse(format, timeStr) if parseErr == nil { break } } if parseErr != nil { // 尝试解析为时间戳字符串 timestamp, err := strconv.ParseInt(timeStr, 10, 64) if err == nil { parsedTime = time.Unix(timestamp, 0) parseErr = nil } else { parseErr = errors.New("failed to parse time") return parsedTime, parseErr } } return parsedTime, nil } |
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
// 生成随机字符串 // 关于随机数种子：https://www.cnblogs.com/niuben/p/13925133.html // 如果在Go语言中使用相同的随机数种子，那么产生的随机数序列将是确定性的，并且每次运行时都会得到相同的结果。 func GetRandomString(length int) string { // 定义用于生成随机字符串的字符集 str := "0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ" // 将字符集转换为字节切片 bytes := []byte(str) // 创建一个空的字节切片以存储结果 result := []byte{} // 使用当前时间作为种子初始化一个新的随机数生成器 r := rand.New(rand.NewSource(time.Now().UnixNano())) // 遍历指定长度以生成随机字符串 for i := 0; i < length; i++ { // 从字符集中随机选择一个字符并追加到结果字节切片中 result = append(result, bytes[r.Intn(len(bytes))]) } // 将最终的字节切片转换为字符串并返回 return string(result) } |
「三年博客，如果觉得我的文章对您有用，请帮助本站成长」
上一篇：golang 实现简易协程池
分别用不同厚度的筏板定义，画图后这设置筏板变截面处理。 http://f.fwxgx.co...
评：新文章！分别用不同厚度的筏板定义，画图后这设置筏板变截面处理。 http://f.fwxgx.co...
评：新文章！新增一个框架图！ http://biji.jinli.vip/wp-content/upl...
评：新文章！嗨，这是一条评论。 要开始审核、编辑及删除评论，请访问仪表盘的“评论”页面。 评论者头像来自...
评：世界，您好！
共有 0 - golang 常用工具方法